UI not updating after updating the state